python list comprehension if statement